package com.wyu.servlet;

import org.apache.commons.io.IOUtils;

import javax.servlet.ServletContext;
import javax.servlet.ServletException;
import javax.servlet.ServletOutputStream;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.net.URLEncoder;

@WebServlet("/Filedownload")
public class Filedownload extends HttpServlet {
    protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {

    }


    protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
//        //1.获取需要下载的文件名
        String downloadFileName = request.getParameter("fileName");

        //2.读取要下载的文件内容(通过 ServletContext 对象可以获取)
        ServletContext servletContext = getServletContext();

        //获取要下载的文件类型
        String mimeType = servletContext.getMimeType("/file/" + downloadFileName);
        System.out.println("下载的文件类型：" + mimeType);

        //4.在回传之前，通过响应头告诉客户端返回的数据类型
        response.setContentType(mimeType);
        //5.还要告诉客户端收到的数据适用于下载使用（还是使用响应头）
        // Content-Disposition 响应头，表示收到的数据怎么处理
        // attachment 表示附件，下载使用；
        // filename= 表示指定下载的文件名
        response.setHeader("Content-Disposition","attachment;filename=" + URLEncoder.encode(downloadFileName,"UTF-8"));

        /*
        在服务器端，第一个 / 被解析为： http://ip:port/工程名/  映射到代码的 web 目录
         */
        InputStream resourceAsStream = servletContext.getResourceAsStream("/file/" + downloadFileName);
        // 原先是用 byte[] 来读，这里改变一下，通过使用 commons-io-1.4.jar 包里已经写好的工具类 IOUtils
        // 获取响应的输出流
        OutputStream outputStream = response.getOutputStream();

        //3.把下载的文件内容传回客户端
        // 读取输出流中全部的数据，复制给输出流，输出给客户端
        IOUtils.copy(resourceAsStream,outputStream);
    }
}
